python quiz code